package com.amigo.online.provider.video.common;

import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.TimeUnit;

import javax.annotation.Resource;

import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.data.redis.core.HashOperations;
import org.springframework.data.redis.core.ListOperations;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.SetOperations;
import org.springframework.data.redis.core.ValueOperations;
import org.springframework.data.redis.core.ZSetOperations;
import org.springframework.lang.Nullable;
import org.springframework.stereotype.Component;

/**
 * @author kml 2018/11/19 读写分离redis Api封装
 */
@SuppressWarnings({"rawtypes", "unchecked"})
@Component
public class RwRedisService {

    @Resource
    @Qualifier("readRedisTemplate")
    RedisTemplate readRedisTemplate;

    @Resource
    @Qualifier("writeRedisTemplate")
    RedisTemplate writeRedisTemplate;

    // Key（键），简单的key-value操作

    /**
     * 实现命令：TTL key，以秒为单位，返回给定 key的剩余生存时间(TTL, time to live)。
     *
     * @param key
     * @return
     */
    public long getExpire(Object key) {
        return readRedisTemplate.getExpire(key);
    }

    /**
     * 实现命令：expire 设置过期时间，单位秒
     *
     * @param key
     * @return
     */
    public void expire(Object key, long timeout) {
        writeRedisTemplate.expire(key, timeout, TimeUnit.SECONDS);
    }

    /**
     * 实现命令：INCR key，增加key一次
     *
     * @param key
     * @return
     */
    public long increment(Object key, long delta) {
        return writeRedisTemplate.opsForValue().increment(key, delta);
    }

    /**
     * 实现命令：KEYS pattern，查找所有符合给定模式 pattern的 key
     */
    public Set<Object> keys(Object pattern) {
        return readRedisTemplate.keys(pattern);
    }

    /**
     * 实现命令：EXISTS key，检查给定 key 是否存在
     *
     * @param key
     */
    public Boolean hasKey(Object key) {
        return writeRedisTemplate.hasKey(key);
    }

    /**
     * 实现命令：DEL key，删除一个key
     *
     * @param key
     */
    public void delete(Object key) {
        writeRedisTemplate.delete(key);
    }

    // String（字符串）

    /**
     * 实现命令：SET key value，设置一个key-value（将字符串值 value关联到 key）
     *
     * @param key
     * @param value
     */
    public void set(Object key, Object value) {
        writeRedisTemplate.opsForValue().set(key, value);
    }

    /**
     * 实现命令：SET key value EX seconds，设置key-value和超时时间（秒）
     *
     * @param key
     * @param value
     * @param timeout （以秒为单位）
     */
    public void set(Object key, Object value, long timeout) {
        writeRedisTemplate.opsForValue().set(key, value, timeout, TimeUnit.SECONDS);
    }

    /**
     * 实现命令：GET key，返回 key所关联的字符串值。
     *
     * @param key
     * @return value
     */
    public Object get(Object key) {
        return readRedisTemplate.opsForValue().get(key);
    }

    // Hash（哈希表）

    /**
     * 实现命令：HSET key field value，将哈希表 key中的域 field的值设为 value
     *
     * @param key
     * @param field
     * @param value
     */
    public void hput(Object key, Object field, Object value) {
        writeRedisTemplate.opsForHash().put(key, field, value);
    }

    /**
     * 实现命令：HSET key field value，将哈希表 key中的域 field的值设为 value
     *
     * @param key
     * @param field
     * @param value
     */
    public void hputAll(Object key, Map map) {
        writeRedisTemplate.opsForHash().putAll(key, map);
    }

    /**
     * 实现命令：HEXISTS key field，查看哈希表 key 中，给定域 field 是否存在
     *
     * @param key
     * @param field
     * @return
     */
    public Boolean hhasKey(String key, Object field) {
        return readRedisTemplate.opsForHash().hasKey(key, field);
    }

    /**
     * 实现命令：HGET key field，返回哈希表 key中给定域 field的值
     *
     * @param key
     * @param field
     * @return
     */
    public Object hget(Object key, Object field) {
        return readRedisTemplate.opsForHash().get(key, field);
    }

    /**
     * 实现命令：HDEL key field [field ...]，删除哈希表 key 中的一个或多个指定域，不存在的域将被忽略。
     *
     * @param key
     * @param fields
     */
    public void hdel(Object key, Object... fields) {
        writeRedisTemplate.opsForHash().delete(key, fields);
    }

    /**
     * 实现命令：HGETALL key，返回哈希表 key中，所有的域和值。
     *
     * @param key
     * @return
     */
    public Map<Object, Object> entries(Object key) {
        return readRedisTemplate.opsForHash().entries(key);
    }

    // List（列表）

    /**
     * 实现命令：LPUSH key value，将一个值 value插入到列表 key的表头
     *
     * @param key
     * @param value
     * @return 执行 LPUSH命令后，列表的长度。
     */
    public Long leftPush(Object key, Object value) {
        return writeRedisTemplate.opsForList().leftPush(key, value);
    }

    /**
     * 实现命令：LPOP key，移除并返回列表 key的头元素。
     *
     * @param key
     * @return 列表key的头元素。
     */
    public Object leftPop(Object key) {
        return writeRedisTemplate.opsForList().leftPop(key);
    }

    /**
     * 实现命令：RPUSH key value，将一个值 value插入到列表 key的表尾(最右边)。
     *
     * @param key
     * @param value
     * @return 执行 LPUSH命令后，列表的长度。
     */
    public Long rightPush(Object key, Object value) {
        return writeRedisTemplate.opsForList().rightPush(key, value);
    }

    /**
     * 实现命令：RPUSH key value，将一个值 value插入到列表 key的表尾(最右边)。
     *
     * @param key
     * @param value
     * @return 执行 LPUSH命令后，列表的长度。
     */
    public Object rightPop(Object key) {
        return writeRedisTemplate.opsForList().rightPop(key);
    }

    /**
     * Get elements between {@code begin} and {@code end} from list at {@code key}.
     *
     * @param key   must not be {@literal null}.
     * @param start
     * @param end
     * @return {@literal null} when used in pipeline / transaction.
     * @see <a href="http://redis.io/commands/lrange">Redis Documentation: LRANGE</a>
     */
    @Nullable
    public List<Object> lrange(Object key, long start, long end) {
        return readRedisTemplate.opsForList().range(key, start, end);
    }

    // Set（集合）

    /**
     * 实现命令：SADD key member [member ...] 将一个或多个 member 元素加入到集合 key 当中，已经存在于集合的
     * member 元素将被忽略
     *
     * @param key
     * @param values
     */
    public void sadd(Object key, Object... values) {
        writeRedisTemplate.opsForSet().add(key, values);
    }

    /**
     * 实现命令：SCARD key [member ...] 返回集合 key 的基数(集合中元素的数量)
     *
     * @param key
     * @param
     * @return 集合中元素的数量
     */
    public Long scard(Object key) {
        return readRedisTemplate.opsForSet().size(key);
    }

    /**
     * 实现命令：SMEMBERS key 返回集合 key 中的所有成员,不存在的 key 被视为空集合
     *
     * @param key
     * @param
     * @return 集合中的所有成员
     */
    public Set<Object> smembers(Object key) {
        return readRedisTemplate.opsForSet().members(key);
    }

    /**
     * 实现命令：SISMEMBER key member 判断 member 元素是否集合 key 的成员
     *
     * @param key
     * @param
     * @return
     */
    public Boolean sisMember(Object key, Object o) {
        return readRedisTemplate.opsForSet().isMember(key, o);
    }

    //SortedSet（有序集合）

    /**
     * Add {@code tuples} to a sorted set at {@code key}, or update its {@code score} if it already exists.
     *
     * @param key    must not be {@literal null}.
     * @param tuples must not be {@literal null}.
     * @return {@literal null} when used in pipeline / transaction.
     * @see <a href="http://redis.io/commands/zadd">Redis Documentation: ZADD</a>
     */
    public Long zadd(Object key, Set tuples) {
        return writeRedisTemplate.opsForZSet().add(key, tuples);
    }

    /**
     * Add {@code value} to a sorted set at {@code key}, or update its {@code score} if it already exists.
     *
     * @param key   must not be {@literal null}.
     * @param score the score.
     * @param value the value.
     * @return {@literal null} when used in pipeline / transaction.
     * @see <a href="http://redis.io/commands/zadd">Redis Documentation: ZADD</a>
     */
    public Boolean zadd(Object key, Object value, double score) {
        return writeRedisTemplate.opsForZSet().add(key, value, score);
    }

    /**
     * Get the size of sorted set with {@code key}.
     *
     * @param key must not be {@literal null}.
     * @return {@literal null} when used in pipeline / transaction.
     * @see <a href="http://redis.io/commands/zcard">Redis Documentation: ZCARD</a>
     * @since 1.3
     */
    @Nullable
    public Long zCard(Object key) {
        return readRedisTemplate.opsForZSet().zCard(key);
    }

    /**
     * Get elements between {@code start} and {@code end} from sorted set.
     *
     * @param key   must not be {@literal null}.
     * @param start
     * @param end
     * @return {@literal null} when used in pipeline / transaction.
     * @see <a href="http://redis.io/commands/zrange">Redis Documentation: ZRANGE</a>
     */
    @Nullable
    public Set<Object> zrange(Object key, long start, long end) {
        return readRedisTemplate.opsForZSet().range(key, start, end);
    }


    public ValueOperations opsForValue() {
        return writeRedisTemplate.opsForValue();
    }

    public ListOperations opsForList() {
        return writeRedisTemplate.opsForList();
    }

    public HashOperations opsForHash() {
        return writeRedisTemplate.opsForHash();
    }

    public SetOperations opsForSet() {
        return writeRedisTemplate.opsForSet();
    }

    public ZSetOperations opsForZSet() {
        return writeRedisTemplate.opsForZSet();
    }

    public ValueOperations opsForValueRead() {
        return readRedisTemplate.opsForValue();
    }

    public ListOperations opsForListRead() {
        return readRedisTemplate.opsForList();
    }

    public HashOperations opsForHashRead() {
        return readRedisTemplate.opsForHash();
    }

    public SetOperations<Object, Object> opsForSetRead() {
        return readRedisTemplate.opsForSet();
    }

    public ZSetOperations<Object, Object> opsForZSetRead() {
        return readRedisTemplate.opsForZSet();
    }
}
